from src.data_preparation import prepare_data
from src.feature_engineering import feature_engineering
from src.model_training import train_models
import time


def main():
    print("Starting training pipeline...")
    start_time = time.time()

    # 数据准备
    print("\n=== Data Preparation ===")
    prepare_data()

    # 特征工程
    print("\n=== Feature Engineering ===")
    feature_engineering()

    # 模型训练
    print("\n=== Model Training ===")
    train_models()

    elapsed = time.time() - start_time
    print(f"\nTraining completed in {elapsed:.2f} seconds")


if __name__ == "__main__":
    main()